package com.coder.base.service;

import com.coder.base.dao.LabelDao;
import com.coder.base.pojo.Label;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import util.IdWorker;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.ArrayList;
import java.util.List;

/**
 * @BelongsProject:coder_parent
 * @BelongsPackage:com.coder.base.service
 * @Author:linchong
 * @CreateTime:2018-12-28 13:47
 * @Description:服务层
 */
@Service
@Transactional
public class LabelService {
    @Autowired
    private LabelDao labelDao;
    @Autowired
    private IdWorker idWorker;


    public List<Label> findAll() {
        return labelDao.findAll();
    }

    public Label findById(String id) {
        return labelDao.findById(id).get();
    }

    public void save(Label label) {
        label.setId(idWorker.nextId() + "");
        labelDao.save(label);
    }

    public void update(Label label) {
        labelDao.save(label);
    }

    public void deleteById(String id) {
        labelDao.deleteById(id);
    }

    /**
     * 根据条件查询标签
     *
     * @param label
     * @return
     */
    public List<Label> findSearch(Label label) {
        //findAll():可以指定参数完成指定的功能，分页，条件查询等
        //传入查询条件
        return labelDao.findAll(new Specification<Label>() {
            /**
             *
             * @param root  根对象，即要把条件封装到那个对象中，类似于where 列名(即类名) = label.getId
             * @param query  封装查询的关键字，如group by ,order by等
             * @param cb    用来封装条件对象，如果直接返回，表示不需要任何条件
             * @return
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                //使用集合存放不确定的条件
                List<Predicate> list = new ArrayList<>();
                if (label.getLabelname() != null && !"".equals(label.getLabelname())) {
                    Predicate predicate = cb.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");
                    //模糊查询，where labelname like %张三%;
                    list.add(predicate);
                }
                if (label.getState() != null && !"".equals(label.getState())) {
                    Predicate predicate = cb.equal(root.get("state").as(String.class), label.getState());
                    //where state = "1";
                    list.add(predicate);
                }
                //new 一个数组作为最终返回值的条件
                Predicate[] parr = new Predicate[list.size()];
                //将list直接转成数组
                list.toArray(parr);
                return cb.and(parr);  //where labelname like "张三" and state = 1;
            }
        });
    }

    /**
     * 分页查询数据
     * @param label
     * @param page
     * @param size
     * @return
     */
    public Page<Label> pageQuery(Label label, int page, int size) {
        //初始化页码，和框架一致，从0页开始，封装分页对象
        Pageable pageable = PageRequest.of(page-1,size);
        return labelDao.findAll(new Specification<Label>() {
            /**
             *
             * @param root  根对象，即要把条件封装到那个对象中，类似于where 列名(即类名) = label.getId
             * @param query  封装查询的关键字，如group by ,order by等
             * @param cb    用来封装条件对象，如果直接返回，表示不需要任何条件
             * @return
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                //使用集合存放不确定的条件
                List<Predicate> list = new ArrayList<>();
                if (label.getLabelname() != null && !"".equals(label.getLabelname())) {
                    Predicate predicate = cb.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");
                    //模糊查询，where labelname like %张三%;
                    list.add(predicate);
                }
                if (label.getState() != null && !"".equals(label.getState())) {
                    Predicate predicate = cb.equal(root.get("state").as(String.class), label.getState());
                    //where state = "1";
                    list.add(predicate);
                }
                //new 一个数组作为最终返回值的条件
                Predicate[] parr = new Predicate[list.size()];
                //将list直接转成数组
                list.toArray(parr);
                return cb.and(parr);  //where labelname like "张三" and state = 1;
            }
        },pageable);
    }
}
